/**
 * Copyright (c) 2008, Andrew Carter All rights reserved.
 * 
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 * 
 * Redistributions of source code must retain the above copyright notice, this
 * list of conditions and the following disclaimer. Redistributions in binary
 * form must reproduce the above copyright notice, this list of conditions and
 * the following disclaimer in the documentation and/or other materials provided
 * with the distribution. Neither the name of Andrew Carter nor the names of
 * contributors may be used to endorse or promote products derived from this
 * software without specific prior written permission.
 * 
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 */
package com.acarter.ardor3d.jtree.treenode;

import javax.swing.Icon;

import com.acarter.ardor3d.jtree.JTreeIcon;
import com.acarter.composabletree.ComposableTreeNode;
import com.ardor3d.renderer.state.RenderState;
import com.ardor3d.renderer.state.RenderState.StateType;
import com.ardor3d.scenegraph.Spatial;

/**
 * @author Carter
 * 
 */
public class RenderStateTreeNode implements ComposableTreeNode {

	/**
	 * @see com.acarter.composabletree.ComposableTreeNode#getChild(java.lang.Object, int)
	 */
	public Object getChild(Object parent, int index) {

		int curIndex = -1;

		if((parent != null) && (parent instanceof Spatial)) {

			for (StateType type : StateType.values()) {

				RenderState state = ((Spatial)parent).getLocalRenderState(type);

				if (state != null) {
					curIndex++;

					if (curIndex == index)
						return state;
				}
			}
		}

		return null;
	}

	/**
	 * @see com.acarter.composabletree.ComposableTreeNode#getChildCount(java.lang.Object)
	 */
	public int getChildCount(Object parent) {

		int count = 0;

		if((parent != null) && (parent instanceof Spatial)) {

			for(StateType state : StateType.values()) {

				if(((Spatial) parent).getLocalRenderState(state) != null)
					count++;
			}
		}

		return count;
	}

	/**
	 * @see com.acarter.composabletree.ComposableTreeNode#getChildIndex(java.lang.Object,
	 *      java.lang.Object)
	 */
	public int getChildIndex(Object parent, Object child) {

		int index = -1;

		if((parent != null) && (parent instanceof Spatial)) {

			Spatial spatial = (Spatial) parent;

			if((child != null) && (child instanceof RenderState)) {
				
				RenderState renderState = (RenderState)child;

				for(StateType state : StateType.values()) {

					if(spatial.getLocalRenderState(state) != null)
						index++;

					if(renderState.getType() == state)
						break;
				}
			}
		}

		return index;
	}

	/**
	 * @see com.acarter.composabletree.ComposableTreeNode#getParent(java.lang.Object)
	 */
	public Object getParent(Object child) {

		// RenderStates are not aware of their parent
		return null;
	}

	/**
	 * @see com.acarter.composabletree.ComposableTreeNode#getNodeClassType()
	 */
	public Class<?> getNodeClassType() {

		return RenderState.class;
	}

	/**
	 * @see com.acarter.composabletree.ComposableTreeNode#getIcon(Object)
	 */
	public Icon getIcon(Object node) {

		if (node instanceof RenderState)  {

			RenderState rs = (RenderState) node;

			if (rs.isEnabled())
				return JTreeIcon.getIcon().renderState;
			else
				return JTreeIcon.getIcon().renderStateDisabled;
		}
		
		return null;
	}

	/**
	 * @see com.acarter.composabletree.ComposableTreeNode#getNodeText(Object, boolean, boolean, boolean, int, boolean)
	 */
	public String getNodeText(Object node, boolean selected, boolean expanded, boolean leaf, int row, boolean hasFocus) {

		if (node instanceof RenderState)
			return ((RenderState) node).getType().toString();
		
		return (node != null) ? node.toString() : "";
	}

	/**
	 * @see com.acarter.composabletree.ComposableTreeNode#getNodeToolTipText(Object, boolean, boolean, boolean, int, boolean)
	 */
	public String getNodeToolTipText(Object node, boolean selected, boolean expanded, boolean leaf, int row, boolean hasFocus) {

		return (node != null) ? node.toString() : "";
	}

}
